Taking inputs in python